#include <stdio.h>      // 引入标准输入输出库
#include <stdlib.h>     // 引入标准库（如 exit 函数）
#include <string.h>     // 引入字符串处理函数库
#include <unistd.h>     // 引入Unix标准函数库（如 close 函数）
#include <arpa/inet.h>  // 引入网络地址转换库（如 inet_addr 函数）
#include <sys/socket.h> // 引入套接字库

#define BUF_SIZE 30 // 定义缓冲区大小

// 错误处理函数，输出错误信息并退出程序
void error_handling(char *message);

int main(int argc, char *argv[]) {
    int sock; // 套接字描述符
    char message[BUF_SIZE]; // 用于存储发送和接收的消息
    int str_len; // 存储接收到的消息长度
    socklen_t adr_sz; // 套接字地址的大小（多余变量）

    struct sockaddr_in serv_adr; // 服务器地址结构体
    struct sockaddr_in from_adr; // 客户端地址结构体（未使用）

    // 检查命令行参数数量
    if (argc != 3) {
        printf("Usage : %s <IP> <port>\n", argv[0]); // 输出用法提示
        exit(1); // 退出程序
    }

    // 创建 UDP 套接字
    sock = socket(PF_INET, SOCK_DGRAM, 0); // PF_INET 表示 IPv4，SOCK_DGRAM 表示 UDP
    if (sock == -1) // 检查套接字创建是否成功
        error_handling("socket() error"); // 调用错误处理函数

    // 初始化服务器地址结构体
    memset(&serv_adr, 0, sizeof(serv_adr)); // 清空结构体
    serv_adr.sin_family = AF_INET; // 设置地址族为 IPv4
    serv_adr.sin_addr.s_addr = inet_addr(argv[1]); // 将输入的 IP 地址转换为网络字节序
    serv_adr.sin_port = htons(atoi(argv[2])); // 将输入的端口号转换为网络字节序

    // 与服务器地址建立连接，UDP 套接字不需要正式连接，但可以简化发送
    connect(sock, (struct sockaddr *) &serv_adr, sizeof(serv_adr));

    // 循环发送和接收消息
    while (1) {
        fputs("Insert message(q to quit): ", stdout); // 提示用户输入消息
        fgets(message, sizeof(message), stdin); // 读取用户输入的消息
        if (!strcmp(message, "q\n") || !strcmp(message, "Q\n")) // 检查是否输入退出指令
            break; // 退出循环

        // 向服务器发送数据
        // 使用 write 发送数据，自动使用已连接的目标地址
//        sendto(sock, message, strlen(message), 0, (struct sockaddr *) &serv_adr, sizeof(serv_adr));
        write(sock, message, strlen(message));

        // 接收来自服务器的消息
//        adr_sz = sizeof(from_adr);
//        str_len = recvfrom(sock, message,BUF_SIZE,0, (struct sockaddr *)&from_adr,&adr_sz);
        str_len = read(sock, message, sizeof(message) - 1); // 读取服务器返回的数据
        message[str_len] = 0; // 在消息末尾添加字符串结束符
        printf("Message from server: %s", message); // 输出接收到的消息
    }

    close(sock); // 关闭套接字
    return 0; // 返回程序正常结束
}

// 错误处理函数实现
void error_handling(char *message) {
    fputs(message, stderr); // 输出错误信息到标准错误输出
    fputc('\n', stderr);    // 输出换行符
    exit(1); // 退出程序
}
